
from model.Siamese import Siamese

def init_model(args):
    Match_Model = Siamese(
        batch_size=args.batch_size,
        channel=args.channel,
        input_shape=args.input_shape,
        backbone=args.backbone,
        backbone_version=args.backbone_version,
        pretrained=args.pretrained,
        distance_type=args.distance_type
    )
    return Match_Model